home *** CD-ROM | disk | FTP | other *** search
/ SGI Freeware 2002 November / SGI Freeware 2002 November - Disc 3.iso / dist / fw_qt3.idb / usr / freeware / Qt / examples / demo / sql / sqlex.ui.h.z / sqlex.ui.h
Encoding:
C/C++ Source or Header  |  2002-04-08  |  5.0 KB  |  150 lines
  1. /****************************************************************************
  2. ** ui.h extension file, included from the uic-generated form implementation.
  3. **
  4. ** If you wish to add, delete or rename slots use Qt Designer which will
  5. ** update this file, preserving your code. Create an init() slot in place of
  6. ** a constructor, and a destroy() slot in place of a destructor.
  7. *****************************************************************************/
  8. #include <qsqldriver.h>
  9. #include <qmessagebox.h>
  10. #include <qsqldatabase.h>
  11. #include <qlineedit.h>
  12. #include <qcombobox.h>
  13. #include <qspinbox.h>
  14. #include <qsqlerror.h>
  15. #include "connect.h"
  16.  
  17. class QCustomSqlCursor: public QSqlCursor
  18. {
  19. public:
  20.     QCustomSqlCursor( const QString & query = QString::null, bool autopopulate = TRUE, QSqlDatabase* db = 0 ) :
  21.         QSqlCursor( QString::null, autopopulate, db )
  22.     {
  23.     exec( query );
  24.     if ( isSelect() && autopopulate ) {
  25.         QSqlRecordInfo inf = ((QSqlQuery*)this)->driver()->recordInfo( *(QSqlQuery*)this );
  26.         for ( QSqlRecordInfo::iterator it = inf.begin(); it != inf.end(); ++it ) {
  27.         append( *it );
  28.         }        
  29.     }
  30.     setMode( QSqlCursor::ReadOnly );
  31.     }
  32.     QCustomSqlCursor( const QCustomSqlCursor & other ): QSqlCursor( other ) {}
  33.     bool select( const QString & /*filter*/, const QSqlIndex & /*sort*/ = QSqlIndex() )
  34.     { return exec( lastQuery() ); }
  35.     QSqlIndex primaryIndex( bool /*prime*/ = TRUE ) const
  36.     { return QSqlIndex(); }
  37.     int insert( bool /*invalidate*/ = TRUE )
  38.     { return FALSE; }
  39.     int update( bool /*invalidate*/ = TRUE )
  40.     { return FALSE; }
  41.     int del( bool /*invalidate*/ = TRUE )
  42.     { return FALSE; }
  43.     void setName( const QString& /*name*/, bool /*autopopulate*/ = TRUE ) {}
  44. };
  45.  
  46. static void showError( const QSqlError& err, QWidget* parent = 0 )
  47. {
  48.    QString errStr ( "The database reported an error\n" );
  49.     if ( !err.databaseText().isEmpty() )
  50.     errStr += err.databaseText();
  51.     if ( !err.driverText().isEmpty() )
  52.     errStr += err.driverText();
  53.     QMessageBox::warning( parent, "Error", errStr );
  54. }
  55. void SqlEx::init()
  56. {
  57.     hsplit->setResizeMode( lv, QSplitter::KeepSize );
  58.     vsplit->setResizeMode( gb, QSplitter::KeepSize );
  59.     submitBtn->setEnabled( FALSE );
  60. }
  61.  
  62. void SqlEx::dbConnect()
  63. { 
  64.     ConnectDialog* conDiag = new ConnectDialog( this, "Connection Dialog", TRUE );
  65.     if ( conDiag->exec() != QDialog::Accepted )
  66.     return;
  67.     if ( dt->sqlCursor() ) {
  68.     dt->setSqlCursor( 0 );
  69.     }
  70.     // close old connection (if any)
  71.     if ( QSqlDatabase::contains( "SqlEx" ) ) {
  72.     QSqlDatabase* oldDb = QSqlDatabase::database( "SqlEx" );
  73.     oldDb->close();
  74.     QSqlDatabase::removeDatabase( "SqlEx" );
  75.     }
  76.     // open the new connection
  77.     QSqlDatabase* db = QSqlDatabase::addDatabase( conDiag->comboDriver->currentText(), "SqlEx" );
  78.     if ( !db ) {
  79.     QMessageBox::warning( this, "Error", "Could not open database" );    
  80.     return;
  81.     }
  82.     db->setHostName( conDiag->editHostname->text() );
  83.     db->setDatabaseName( conDiag->editDatabase->text() );
  84.     db->setPort( conDiag->portSpinBox->value() );
  85.     if ( !db->open( conDiag->editUsername->text(), conDiag->editPassword->text() ) ) {
  86.     showError( db->lastError(), this );
  87.     return;
  88.     }
  89.     lbl->setText( "Double-Click on a table-name to view the contents" );
  90.     lv->clear();
  91.     
  92.     QStringList tables = db->tables();
  93.     for ( QStringList::Iterator it = tables.begin(); it != tables.end(); ++it ) {
  94.     QListViewItem* lvi = new QListViewItem( lv, *it );
  95.     QSqlRecordInfo ri = db->recordInfo ( *it );
  96.     for ( QSqlRecordInfo::Iterator it = ri.begin(); it != ri.end(); ++it ) {
  97.         QString req;
  98.         if ( (*it).isRequired() > 0 ) {
  99.         req = "Yes";
  100.         } else if ( (*it).isRequired() == 0 ) {
  101.         req = "No";
  102.         } else {
  103.         req = "?";
  104.         }
  105.         QListViewItem* fi = new QListViewItem( lvi, (*it).name(),  + QVariant::typeToName( (*it).type() ), req );
  106.         lvi->insertItem( fi );
  107.     }
  108.     lv->insertItem( lvi );    
  109.     }
  110.     submitBtn->setEnabled( TRUE );
  111. }
  112.  
  113. void SqlEx::execQuery()
  114. {
  115.     // use a custom cursor to populate the data table
  116.     QCustomSqlCursor* cursor = new QCustomSqlCursor( te->text(), TRUE, QSqlDatabase::database( "SqlEx", TRUE ) );
  117.     if ( cursor->isSelect() ) {
  118.     dt->setSqlCursor( cursor, TRUE, TRUE );
  119.     dt->refresh( QDataTable::RefreshAll );
  120.     QString txt( "Query OK" );
  121.     if ( cursor->size() >= 0 )
  122.         txt += ", returned rows: " + QString::number( cursor->size() );
  123.     lbl->setText( txt );
  124.     } else {
  125.     // an error occured if the cursor is not active
  126.     if ( !cursor->isActive() ) {
  127.         showError( cursor->lastError(), this );
  128.     } else {
  129.         lbl->setText( QString("Query OK, affected rows: %1").arg( cursor->numRowsAffected() ) );
  130.     }
  131.     }
  132. }
  133.  
  134. void SqlEx::showTable( QListViewItem * item )
  135. {
  136.     // get the table name
  137.     QListViewItem* i = item->parent();
  138.     if ( !i ) {
  139.     i = item;
  140.     }
  141.  
  142.     // populate the data table
  143.     QSqlCursor* cursor = new QSqlCursor( i->text( 0 ), TRUE, QSqlDatabase::database( "SqlEx", TRUE ) );
  144.     dt->setSqlCursor( cursor, TRUE, TRUE );
  145.     dt->setSort( cursor->primaryIndex() );
  146.     dt->refresh( QDataTable::RefreshAll );
  147.     lbl->setText( "Displaying table " + i->text( 0 ) );
  148. }
  149.  
  150.